package com.yaoqiang;

import org.apache.zookeeper.*;

import java.io.IOException;

/**
 * @ClassName ZkClusterDemo
 * @Description TODO
 * @Author yaoqiang
 * @Date 2020/11/27 15:22
 * @Version 1.0
 */
public class ZkClusterDemo {
    // 会话超时时间， 设置为与系统默认时间一致
    private static final int SESSION_TIMEOUT = 30 * 1000;

    // 创建zookeeper实例
    private ZooKeeper zk;

    // 创建Watcher实例
    private Watcher wh = new Watcher() {

        /**
         * Watched事件
         */
        public void process(WatchedEvent event) {
            System.out.println("WatchedEvent >>> " + event.toString());
        }
    };

    // 初始化zookeeper实例
    @SuppressWarnings("static-access")
    private void createZKInstance() throws Exception {
        // 连接到zk服务，多个服务可以用,分割写
        zk = new ZooKeeper("192.168.55.214:3000,192.168.55.214:3001,192.168.55.214:3002", SESSION_TIMEOUT, this.wh);
    }

    // zookeeper 操作
    private void ZKOperations() throws IOException, InterruptedException, KeeperException {
        System.out.println("\n1. 创建Zookeeper 节点(znode : zoo2, 数据: myData2, 权限: OPEN_ACL_UNSAFE, 节点类型: Persistent.)");
        zk.create("/zoo2", "myData2".getBytes(), ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);

        System.out.println("\n2. 查看是否创建成功:");
        System.out.println(new String(zk.getData("/zoo2", this.wh, null)));// 添加Watcher,设置下次修改监控

        // 前面一行我们添加了对/zoo2节点的监视，所以这里对/zoo2进行修改的时候，会触发Watch时间
        System.out.println("\n3. 修改/zoo2节点数据");
        zk.setData("/zoo2", "helloword".getBytes(), -1);

        // 这里再次进行修改，则不会触发Watch事件，这就是我们验证ZK的一个特性“一次触发”，也就是说设置一次监控，只会对下次操作起一次作用。
        System.out.println("\n3-1. 再次修改/zoo2节点数据");
        zk.setData("/zoo2", "helloword-ABCD".getBytes(), -1);

        // 查看
        System.out.println("\n4. 查看是否修改成功:");
        System.out.println(new String(zk.getData("/zoo2", false, null)));// 本次不再设置Watch监控

        // 删除节点
        System.out.println("\n5. 删除节点");
        zk.delete("/zoo2", -1);

        System.out.println("\n6. 查看节点是否被删除:");
        System.out.println("节点状态: [" + zk.exists("/zoo2", false) + " ]" );
    }

    // 关闭zookeeper连接
    @SuppressWarnings("unused")
    private void ZKClose() throws InterruptedException {
        if(zk.getState() == ZooKeeper.States.CONNECTED){
            zk.close();
        }
    }

    public static void main(String[] args) throws Exception {
        ZkClusterDemo zkDemo = new ZkClusterDemo();
        zkDemo.createZKInstance();
        zkDemo.ZKOperations();
        zkDemo.ZKClose();
    }

}
